package com.itcast.yygh.cmn.easyExcel;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.itcast.yygh.cmn.entity.Stu;

import java.util.ArrayList;
import java.util.List;

/**
 * 写入excel表数据的测试方法一
 */

public class EasyExcelWriteTest {
    public static void main(String[] args) {
        //指定写入文件的路径
        String fileName = "D:\\112.xlsx";
        //第一种方式
        //写入文件调用write方法  write方法中需要提供两个参数:1.写入的文件名称 2.提供要写入表头的实体对象
        //sheet:默认写入到excel表中第一个sheet  需要指定sheet的名称
        //doWrite：要写入的数据
        //EasyExcel.write(fileName, Stu.class).sheet("写入方法一").doWrite(data());
        //第二种方式
        ExcelWriter build = EasyExcel.write(fileName, Stu.class).build();
        WriteSheet writeSheet = EasyExcel.writerSheet("写入方法二").build();
        build.write(data(),writeSheet);
        build.finish();
    }
    //循环设置要添加的数据，最终封装到list集合中
    private static List<Stu> data() {
        List<Stu> list = new ArrayList<Stu>();
        for (int i = 0; i < 10; i++) {
            Stu data = new Stu();
            data.setSno(i);
            data.setSname("张三"+i);
            list.add(data);
        }
        return list;
    }
}
